tensor([[0.7800, 0.7800, 0.7800, 0.7800, 0.7800],
        [0.2200, 0.2200, 0.2200, 0.2200, 0.2200]], grad_fn=<RoundBackward1>)